import com.fasterxml.jackson.databind.ObjectMapper;
import com.mysql.jdbc.jdbc2.optional.MysqlDataSource;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.sql.DataSource;
import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

/**
 * Created with IntelliJ IDEA.
 * Description:
 * User: 言初
 * Date: 2024-04-02
 * Time: 13:02
 */
class Message {
    public String from;
    public String to;
    public String message;

    @Override
    public String toString() {
        return "Message{" +
                "from='" + from + '\'' +
                ", to='" + to + '\'' +
                ", message='" + message + '\'' +
                '}';
    }
}

@WebServlet("/message")
public class messageServlet extends HttpServlet {

    private ObjectMapper objectMapper = new ObjectMapper();
    // private List<Message> list = new ArrayList<>();

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        // 通过这个方法来处理获取所有的留言消息这个功能
        // 需要返回一个 json字符串数组，很幸运jackson已经帮我们处理好了格式，
        // 使用下面的这个方法就可以把 咱们的 message 的list 直接转成一个字符串返回了
        List<Message> list = load();
        String respString = objectMapper.writeValueAsString(list);
        resp.setContentType("application/json; charset=utf8");
        resp.getWriter().write(respString);
    }



    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        // 用这个方法来处理 提交新消息这个功能
        // 解析body对象，
        Message message = objectMapper.readValue(req.getInputStream(),Message.class);
        // 把提交的内容给他存起来（存到 list中）
        // list.add(message);
        save(message);
        System.out.println("消息提交成功 "+ message.toString());
        // 默认会返回一个200 的报文
    }


    // 这个方法用来从数据库中查询出所有的结果
    public List<Message> load(){
        List<Message> messageList = new ArrayList<>();

        // 从数据库中获取到 数据 们并返回
        // 设置数据源
        DataSource dataSource = new MysqlDataSource();
        ((MysqlDataSource)dataSource).setUrl("jdbc:mysql://127.0.0.1:3306/java107?characterEncoding=utf8&useSSL=false");
        ((MysqlDataSource)dataSource).setUser("root");
        ((MysqlDataSource)dataSource).setPassword("813818");
        // 建立连接
        try{
            Connection connection = dataSource.getConnection();
            // 是java.sql 的 Connection
            String sql = "select * from message";
            PreparedStatement statement = connection.prepareStatement(sql);

            ResultSet resultSet = statement.executeQuery();

            // 得到之后，还需要去遍历这个 resultSet
            while(resultSet.next()){
                Message message = new Message();
                message.from = resultSet.getString("from");
                message.to = resultSet.getString("to");
                message.message = resultSet.getString("message");
                messageList.add(message);
            }


            resultSet.close();
            statement.close();
            connection.close();

        }catch(SQLException e){
            e.printStackTrace();
        }

        return messageList;
    }

     // 这个方法用来往数据库中存一条数据
    public void save(Message message){
        // 往数据库中存数据
        // 设置数据源
        DataSource dataSource = new MysqlDataSource();
        ((MysqlDataSource)dataSource).setUrl("jdbc:mysql://127.0.0.1:3306/java107?characterEncoding=utf8&useSSL=false");
        ((MysqlDataSource)dataSource).setUser("root");
        ((MysqlDataSource)dataSource).setPassword("813818");
        // 建立连接
        try{
            Connection connection = dataSource.getConnection();
            // 是java.sql 的 Connection
            String sql = "insert into message values(?,?,?)";
            PreparedStatement statement = connection.prepareStatement(sql);
            statement.setString(1,message.from);
            statement.setString(2,message.to);
            statement.setString(3,message.message);

            statement.executeUpdate();

            statement.close();
            connection.close();

        }catch(SQLException e){
            e.printStackTrace();
        }
    }
}
